package com.genius.workflow.util;

import java.io.OutputStream;

import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFCellStyle;
import org.apache.poi.hssf.usermodel.HSSFFont;
import org.apache.poi.hssf.usermodel.HSSFRichTextString;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.BorderStyle;
import org.apache.poi.ss.usermodel.CellType;
import org.apache.poi.ss.usermodel.HorizontalAlignment;
import org.apache.poi.ss.usermodel.IndexedColors;
import org.apache.poi.ss.usermodel.VerticalAlignment;
import org.apache.poi.ss.util.CellRangeAddress;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;

public class ExportExcel {

  // 显示的导出表的标题
  private String title;
  // 导出表的列名
  private String[] rowName;
  private String[] rowField;
  private JSONArray dataList ;

  // 构造函数，传入要导出的数据
  public ExportExcel(String title, String[] rowName, String[] rowField, JSONArray dataList) {
    this.dataList = dataList;
    this.rowName = rowName;
    this.rowField = rowField;
    this.title = title;
  }

  // 导出数据
  public void export(OutputStream out) throws Exception {
    try {
      HSSFWorkbook workbook = new HSSFWorkbook();
      HSSFSheet sheet = workbook.createSheet(title);

      // 产生表格标题行
      HSSFRow rowm = sheet.createRow(0);
      HSSFCell cellTitle = rowm.createCell(0);
      
      
      //sheet样式定义【】
      HSSFCellStyle columnTopStyle=this.getColumnTopStyle(workbook);
      HSSFCellStyle style=this.getStyle(workbook);
      sheet.addMergedRegion(new CellRangeAddress(0, 1, 0, (rowName.length - 1)));
      cellTitle.setCellStyle(columnTopStyle);
      cellTitle.setCellValue(title);

      // 定义所需列数
      int columnNum = rowName.length;
      HSSFRow rowRowName = sheet.createRow(2);

      // 将列头设置到sheet的单元格中
      for (int n = 0; n < columnNum; n++) {
        HSSFCell cellRowName = rowRowName.createCell(n);
        cellRowName.setCellType(CellType.STRING);
        HSSFRichTextString text = new HSSFRichTextString(rowName[n]);
        cellRowName.setCellValue(text);
        cellRowName.setCellStyle(columnTopStyle);
      }
      // 将查询到的数据设置到sheet对应的单元格中
      for (int i = 0; i < dataList.size(); i++) {
        //Object[] obj = dataList.get(i);// 遍历每个对象
        JSONObject obj = dataList.getJSONObject(i);
        HSSFRow row = sheet.createRow(i + 3);// 创建所需的行数
        int j = 0;
        for(String key:rowField){
        	HSSFCell cell = null;
        	cell = row.createCell(j, CellType.STRING);
            if (!"".equals(obj.get(key)) && obj.get(key) != null) {
              cell.setCellValue(obj.get(key).toString());
            }else{
            	cell.setCellValue("");
            }
              cell.setCellStyle(style);
              j++;
        }
      }
      // 让列宽随着导出的列长自动适应
      for (int colNum = 0; colNum < columnNum; colNum++) {
        int columnWidth = sheet.getColumnWidth(colNum) / 256;
        for (int rowNum = 0; rowNum < sheet.getLastRowNum(); rowNum++) {
          HSSFRow currentRow;
          if (sheet.getRow(rowNum) == null) {
            currentRow = sheet.createRow(rowNum);
          } else {
            currentRow = sheet.getRow(rowNum);
          }
          if (currentRow.getCell(colNum) != null) {
            HSSFCell currentCell = currentRow.getCell(colNum);
            if (currentCell.getCellType() == CellType.STRING) {
              int length = currentCell.getStringCellValue().getBytes().length;
              if (columnWidth < length) {
                columnWidth = length;
              }
            }
          }
        }
//        if (colNum == 0) {
//          sheet.setColumnWidth(colNum, (columnWidth - 2) * 256);
//        } else {
//          sheet.setColumnWidth(colNum, (columnWidth + 4) * 256);
//        }
      }

      if (workbook != null) {
        try {

          workbook.write(out);

        } catch (Exception e) {
          e.printStackTrace();
        }
      }

    } catch (Exception e) {
    	e.printStackTrace();
    }
  }
  /*
   * 列头单元格样式
   */
  public HSSFCellStyle getColumnTopStyle(HSSFWorkbook workbook) {
    // 设置字体
    HSSFFont font = workbook.createFont();

    // 设置字体大小
    font.setFontHeightInPoints((short) 11);
    // 字体加粗
//    font.setBold(true);
//    font.setBoldweight(HSSFFont.SS_SUB);
    // 设置字体名字
    font.setFontName("Courier New");
    // 设置样式
    HSSFCellStyle style = workbook.createCellStyle();
    // 设置低边框
    style.setBorderBottom(BorderStyle.THIN);
    // 设置低边框颜色
    style.setBottomBorderColor(IndexedColors.BLACK.getIndex());
    // 设置右边框
    style.setBorderRight(BorderStyle.THIN);
    // 设置顶边框
    style.setTopBorderColor(IndexedColors.BLACK.getIndex());
    // 设置顶边框颜色
    style.setTopBorderColor(IndexedColors.BLACK.getIndex());
    // 在样式中应用设置的字体
    style.setFont(font);
    // 设置自动换行
    style.setWrapText(false);
    // 设置水平对齐的样式为居中对齐；
    style.setAlignment(HorizontalAlignment.CENTER);
    style.setVerticalAlignment(VerticalAlignment.CENTER);
    return style;

  }

  public HSSFCellStyle getStyle(HSSFWorkbook workbook) {
    // 设置字体
    HSSFFont font = workbook.createFont();
    // 设置字体大小
    font.setFontHeightInPoints((short) 10);
    // 字体加粗
//    font.setBold(true);
    // 设置字体名字
    font.setFontName("Courier New");
    // 设置样式;
    HSSFCellStyle style = workbook.createCellStyle();
    // 设置底边框;
    style.setBorderBottom(BorderStyle.THIN);
    // 设置底边框颜色;
    style.setBorderBottom(BorderStyle.THIN);
    // 设置低边框颜色
    style.setBottomBorderColor(IndexedColors.BLACK.getIndex());
    // 设置右边框
    style.setBorderRight(BorderStyle.THIN);
    // 设置顶边框
    style.setTopBorderColor(IndexedColors.BLACK.getIndex());
    // 设置顶边框颜色
    style.setTopBorderColor(IndexedColors.BLACK.getIndex());
    // 在样式中应用设置的字体
    style.setFont(font);
    // 设置自动换行
    style.setWrapText(false);
    // 设置水平对齐的样式为居中对齐；
    style.setAlignment(HorizontalAlignment.CENTER);
    style.setVerticalAlignment(VerticalAlignment.CENTER);
    return style;
  }
}
